package com.gao.masterslave.config;

import com.fasterxml.jackson.annotation.JsonAutoDetect;
import com.fasterxml.jackson.annotation.PropertyAccessor;
import com.fasterxml.jackson.databind.ObjectMapper;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.cache.annotation.CachingConfigurerSupport;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.data.redis.connection.RedisConnectionFactory;
import org.springframework.data.redis.connection.lettuce.LettuceConnectionFactory;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.serializer.Jackson2JsonRedisSerializer;
import org.springframework.data.redis.serializer.StringRedisSerializer;

@Configuration
public class RedisConfig extends CachingConfigurerSupport {

    @Autowired
    private RedisPropertiesConfig redisPropertiesConfig;

    @Bean(name = "masterRedisTemplate")
    @Primary
    public RedisTemplate<String, Object> redisTemplate(@Qualifier("masterConnectionFactory")
                                                                   RedisConnectionFactory masterLettuceConnectionFactory) {
        return setRedisConnectionFactory(masterLettuceConnectionFactory);
    }

    @Bean(name = "slaveRedisTemplate")
    public RedisTemplate<String, Object> slaveRedisTemplate(@Qualifier("slaveConnectionFactory")
                                                                        RedisConnectionFactory slaveLettuceConnectionFactory) {
        return setRedisConnectionFactory(slaveLettuceConnectionFactory);
    }

    private RedisTemplate<String, Object> setRedisConnectionFactory(RedisConnectionFactory redisConnectionFactory) {
        RedisTemplate<String, Object> template = new RedisTemplate<>();
        // 配置连接工厂
        template.setConnectionFactory(redisConnectionFactory);
        //使用Jackson2JsonRedisSerializer来序列化和反序列化redis的value值（默认使用JDK的序列化方式）
        Jackson2JsonRedisSerializer<Object> jacksonSeial = new Jackson2JsonRedisSerializer<>(Object.class);
        ObjectMapper obm = new ObjectMapper();
        // 指定要序列化的域，field,get和set,以及修饰符范围，ANY是都有包括private和public
        obm.setVisibility(PropertyAccessor.ALL, JsonAutoDetect.Visibility.ANY);
        // 指定序列化输入的类型，类必须是非final修饰的，final修饰的类，比如String,Integer等会抛出异常
        obm.enableDefaultTyping(ObjectMapper.DefaultTyping.NON_FINAL);
        jacksonSeial.setObjectMapper(obm);

        // 值采用json序列化
        template.setValueSerializer(jacksonSeial);
        //使用StringRedisSerializer来序列化和反序列化redis的key值
        template.setKeySerializer(new StringRedisSerializer());

        // 设置hash key 和value序列化模式
        template.setHashKeySerializer(new StringRedisSerializer());
        template.setHashValueSerializer(jacksonSeial);
        template.afterPropertiesSet();

        return template;
    }

    /**
     * 配置 redis 主节点连接工厂
     * 需要添加 @Primary 指定bean的名称
     */
    @Bean(name = "masterConnectionFactory")
    @Primary
    public RedisConnectionFactory masterLettuceConnectionFactory() {
        LettuceConnectionFactory lettuceConnectionFactory = new LettuceConnectionFactory();
        lettuceConnectionFactory.setDatabase(redisPropertiesConfig.getmDatabase());
        lettuceConnectionFactory.setHostName(redisPropertiesConfig.getmHost());
        lettuceConnectionFactory.setPort(redisPropertiesConfig.getmPort());
        lettuceConnectionFactory.setPassword(redisPropertiesConfig.getmPassword());
        lettuceConnectionFactory.setTimeout(redisPropertiesConfig.getmTimeout());
        return lettuceConnectionFactory;
    }

    /**
     * 配置 redis 从节点连接工厂
     */
    @Bean(name = "slaveConnectionFactory")
    public RedisConnectionFactory slaveLettuceConnectionFactory() {
        LettuceConnectionFactory lettuceConnectionFactory = new LettuceConnectionFactory();
        lettuceConnectionFactory.setDatabase(redisPropertiesConfig.getsDatabase());
        lettuceConnectionFactory.setHostName(redisPropertiesConfig.getsHost());
        lettuceConnectionFactory.setPort(redisPropertiesConfig.getsPort());
        lettuceConnectionFactory.setPassword(redisPropertiesConfig.getsPassword());
        lettuceConnectionFactory.setTimeout(redisPropertiesConfig.getsTimeout());
        return lettuceConnectionFactory;
    }
}
